package com.abc.util;

import com.abc.config.ChatMemoryConfig;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

import java.util.List;
import java.util.concurrent.TimeUnit;


@Component
public class RedisUtil {

    @Autowired
    private RedisTemplate<String, Object> redisTemplate;
    
    @Autowired
    private ChatMemoryConfig chatMemoryConfig;
    
    // ================================ 常量定义 ================================
    
    /**
     * 聊天消息Redis key前缀
     */
    private static final String CHAT_MESSAGES_KEY_PREFIX = "chat:messages:";
    
    /**
     * 生成聊天消息的Redis key
     *
     * @param conversationId 对话ID
     * @return Redis key
     */
    private String buildChatMessageModelKey(String conversationId) {
        return CHAT_MESSAGES_KEY_PREFIX + conversationId;
    }

    /**
     * 设置键值对
     *
     * @param key   键
     * @param value 值
     */
    public void set(String key, Object value) {
        redisTemplate.opsForValue().set(key, value);
    }

    /**
     * 设置键值对，带过期时间
     *
     * @param key      键
     * @param value    值
     * @param timeout  过期时间
     * @param timeUnit 时间单位
     */
    public void set(String key, Object value, long timeout, TimeUnit timeUnit) {
        redisTemplate.opsForValue().set(key, value, timeout, timeUnit);
    }

    /**
     * 获取值
     *
     * @param key 键
     * @return 值
     */
    public Object get(String key) {
        return redisTemplate.opsForValue().get(key);
    }

    /**
     * 获取值并转换为指定类型
     *
     * @param key   键
     * @param clazz 类型
     * @param <T>   泛型
     * @return 值
     */
    @SuppressWarnings("unchecked")
    public <T> T get(String key, Class<T> clazz) {
        Object value = redisTemplate.opsForValue().get(key);
        if (value != null && clazz.isInstance(value)) {
            return (T) value;
        }
        return null;
    }

    /**
     * 删除键
     *
     * @param key 键
     * @return 是否删除成功
     */
    public Boolean delete(String key) {
        return redisTemplate.delete(key);
    }

    /**
     * 判断键是否存在
     *
     * @param key 键
     * @return 是否存在
     */
    public Boolean hasKey(String key) {
        return redisTemplate.hasKey(key);
    }

    /**
     * 设置过期时间
     *
     * @param key     键
     * @param timeout 过期时间
     * @param unit    时间单位
     * @return 是否设置成功
     */
    public Boolean expire(String key, long timeout, TimeUnit unit) {
        return redisTemplate.expire(key, timeout, unit);
    }

    /**
     * 获取过期时间
     *
     * @param key 键
     * @return 过期时间（秒）
     */
    public Long getExpire(String key) {
        return redisTemplate.getExpire(key);
    }

    /**
     * 递增
     *
     * @param key 键
     * @return 递增后的值
     */
    public Long increment(String key) {
        return redisTemplate.opsForValue().increment(key);
    }

    /**
     * 递增指定值
     *
     * @param key   键
     * @param delta 递增的值
     * @return 递增后的值
     */
    public Long increment(String key, long delta) {
        return redisTemplate.opsForValue().increment(key, delta);
    }

    /**
     * 递减
     *
     * @param key 键
     * @return 递减后的值
     */
    public Long decrement(String key) {
        return redisTemplate.opsForValue().decrement(key);
    }

    /**
     * 递减指定值
     *
     * @param key   键
     * @param delta 递减的值
     * @return 递减后的值
     */
    public Long decrement(String key, long delta) {
        return redisTemplate.opsForValue().decrement(key, delta);
    }

    // ================================ Hash操作 ================================

    /**
     * 设置Hash字段值
     *
     * @param key   键
     * @param field 字段
     * @param value 值
     */
    public void hSet(String key, String field, Object value) {
        redisTemplate.opsForHash().put(key, field, value);
    }

    /**
     * 获取Hash字段值
     *
     * @param key   键
     * @param field 字段
     * @return 值
     */
    public Object hGet(String key, String field) {
        return redisTemplate.opsForHash().get(key, field);
    }

    /**
     * 删除Hash字段
     *
     * @param key    键
     * @param fields 字段数组
     * @return 删除的字段数量
     */
    public Long hDelete(String key, Object... fields) {
        return redisTemplate.opsForHash().delete(key, fields);
    }

    /**
     * 判断Hash字段是否存在
     *
     * @param key   键
     * @param field 字段
     * @return 是否存在
     */
    public Boolean hHasKey(String key, String field) {
        return redisTemplate.opsForHash().hasKey(key, field);
    }

    // ================================ List操作 ================================

    /**
     * 从左边推入元素
     *
     * @param key   键
     * @param value 值
     * @return 列表长度
     */
    public Long lPush(String key, Object value) {
        return redisTemplate.opsForList().leftPush(key, value);
    }

    /**
     * 从右边推入元素
     *
     * @param key   键
     * @param value 值
     * @return 列表长度
     */
    public Long rPush(String key, Object value) {
        return redisTemplate.opsForList().rightPush(key, value);
    }

    /**
     * 从左边弹出元素
     *
     * @param key 键
     * @return 元素
     */
    public Object lPop(String key) {
        return redisTemplate.opsForList().leftPop(key);
    }

    /**
     * 从右边弹出元素
     *
     * @param key 键
     * @return 元素
     */
    public Object rPop(String key) {
        return redisTemplate.opsForList().rightPop(key);
    }

    /**
     * 获取列表长度
     *
     * @param key 键
     * @return 长度
     */
    public Long lSize(String key) {
        return redisTemplate.opsForList().size(key);
    }
    
    /**
     * 批量从左边推入元素
     *
     * @param key    键
     * @param values 值列表
     * @return 列表长度
     */
    public Long lPush(String key, List<Object> values) {
        return redisTemplate.opsForList().leftPushAll(key, values);
    }
    
    /**
     * 获取列表指定范围的元素
     *
     * @param key   键
     * @param start 开始索引
     * @param end   结束索引
     * @return 元素列表
     */
    public List<Object> lRange(String key, long start, long end) {
        return redisTemplate.opsForList().range(key, start, end);
    }
    
    /**
     * 获取列表指定范围的元素并转换为指定类型
     *
     * @param key   键
     * @param start 开始索引
     * @param end   结束索引
     * @param clazz 类型
     * @param <T>   泛型
     * @return 元素列表
     */
    @SuppressWarnings("unchecked")
    public <T> List<T> lRange(String key, long start, long end, Class<T> clazz) {
        List<Object> objects = redisTemplate.opsForList().range(key, start, end);
        if (objects == null || objects.isEmpty()) {
            return new java.util.ArrayList<>();
        }
        return objects.stream()
                .filter(obj -> obj != null && clazz.isInstance(obj))
                .map(obj -> (T) obj)
                .collect(java.util.stream.Collectors.toList());
    }
    
    /**
     * 裁剪列表，保留指定范围的元素
     *
     * @param key   键
     * @param start 开始索引
     * @param end   结束索引
     */
    public void lTrim(String key, long start, long end) {
        redisTemplate.opsForList().trim(key, start, end);
    }
    
    /**
     * 从右边推入元素，并保持列表最大长度（滑动窗口）
     *
     * @param key       键
     * @param value     值
     * @param maxLength 最大长度
     * @return 列表长度
     */
    public Long rPushAndTrim(String key, Object value, long maxLength) {
        Long length = redisTemplate.opsForList().rightPush(key, value);
        // 如果超过最大长度，从左边删除多余元素
        if (length != null && length > maxLength) {
            redisTemplate.opsForList().trim(key, length - maxLength, -1);
        }
        return length;
    }
    
    /**
     * 批量从右边推入元素，并保持列表最大长度（滑动窗口）
     *
     * @param key       键
     * @param values    值列表
     * @param maxLength 最大长度
     * @return 列表长度
     */
    public Long rPushAllAndTrim(String key, List<Object> values, long maxLength) {
        Long length = redisTemplate.opsForList().rightPushAll(key, values);
        // 如果超过最大长度，从左边删除多余元素
        if (length != null && length > maxLength) {
            redisTemplate.opsForList().trim(key, length - maxLength, -1);
        }
        return length;
    }

    // ================================ Set操作 ================================

    /**
     * 添加元素到Set
     *
     * @param key    键
     * @param values 值数组
     * @return 添加的元素数量
     */
    public Long sAdd(String key, Object... values) {
        return redisTemplate.opsForSet().add(key, values);
    }

    /**
     * 从Set中移除元素
     *
     * @param key    键
     * @param values 值数组
     * @return 移除的元素数量
     */
    public Long sRemove(String key, Object... values) {
        return redisTemplate.opsForSet().remove(key, values);
    }

    /**
     * 判断元素是否在Set中
     *
     * @param key   键
     * @param value 值
     * @return 是否存在
     */
    public Boolean sIsMember(String key, Object value) {
        return redisTemplate.opsForSet().isMember(key, value);
    }

    /**
     * 获取Set大小
     *
     * @param key 键
     * @return 大小
     */
    public Long sSize(String key) {
        return redisTemplate.opsForSet().size(key);
    }
    

}
